package xiaohong.algorithm;

import java.util.Arrays;

public class Solution {

    public static void main(String[] args) {
        int[] nums1 = new int[]{1,2,3,0,0,0};
        int[] nums2 = new int[]{2,5,6};
//        merge(nums1 , 3 , nums2 , 3 );
        merge2(nums1 , 3 , nums2 , 3 );
    }

    /*
     * 给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，
     * 另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
     * 请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。
     *
     * 此时 n ！= 0   ， m ！= 0
     * nums1 and nums2 有数据
     * nums1 后面数据为0补齐
     * 最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。
     * 为了应对这种情况，nums1 的初始长度为 m + n，
     * 其中前 m 个元素表示应合并的元素，
     * 后 n 个元素为 0 ，应忽略。
     * nums2 的长度为 n
     */
    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        int[] temp = new int[m + n];
        int i = 0, j = 0, k = 0;
        // 合并两个数组
        while (i < m && j < n) {
            if (nums1[i] <= nums2[j]) {
                temp[k++] = nums1[i++];
            } else {
                temp[k++] = nums2[j++];
            }
        }
        // 复制剩余元素
        while (i < m) {
            temp[k++] = nums1[i++];
        }

        while (j < n) {
            temp[k++] = nums2[j++];
        }
        System.arraycopy(temp, 0, nums1, 0, m + n);
        System.out.println(Arrays.toString(nums1));
    }

    public static void merge2(int[] nums1, int m, int[] nums2, int n) {
        int[] nums3 = new int[m + n];
        int p1 = 0;
        int p2 = 0;
        int p3 = 0;
        while(p1 < m && p2 < n) {
            if(nums1[p1] < nums2[p2]) {
                nums3[p3++] = nums1[p1++];
            } else {
                nums3[p3++] = nums2[p2++];
            }
        }
        if(p1 == m) {
            while(p2 < n) {
                nums3[p3++] = nums2[p2++];
            }
        }
        if(p2 == n) {
            while(p1 < m) {
                nums3[p3++] = nums1[p1++];
            }
        }
        for(int j = 0; j < m + n; j++) {
            nums1[j] = nums3[j];
        }
    }

}